package com.yeci.common;

import lombok.Data;

import java.util.HashMap;

/**
 * @description: 自定义返回格式
 * @author: yeci
 * @date: 2021/8/18 9:40
 **/
@Data
public class AjaxResult extends HashMap<String , Object> {

  private static final long serialVersionUID = 1L;

  /** 状态码 */
  public static final String CODE_TAG = "code";

  /** 返回内容 */
  public static final String MSG_TAG = "msg";

  /** 数据对象 */
  public static final String DATA_TAG = "data";

  public static final String DATA_COUNT = "total";

  /**
   * 初始化一个新创建的 AjaxResult 对象，使其表示一个空消息。
   */
  public AjaxResult() {
  }

  /**
   * 初始化一个新创建的 AjaxResult 对象
   *
   * @param code 状态码
   * @param msg 返回内容
   */
  public AjaxResult(int code, String msg) {
    super.put(CODE_TAG, code);
    super.put(MSG_TAG, msg);
  }

  /**
   * 初始化一个新创建的 AjaxResult 对象
   *
   * @param code 状态码
   * @param msg 返回内容
   * @param data 数据对象
   */
  public AjaxResult(int code, String msg, Object data) {
    super.put(CODE_TAG, code);
    super.put(MSG_TAG, msg);
    super.put(DATA_TAG, data);
  }

  /**
   * 初始化一个新创建的 AjaxResult 对象
   *
   * @param code 状态码
   * @param msg 返回内容
   * @param data 数据对象
   */
  public AjaxResult(int code, String msg, Object data,Object total) {
    super.put(CODE_TAG, code);
    super.put(MSG_TAG, msg);
    super.put(DATA_TAG, data);
    super.put(DATA_COUNT,total);
  }

  /**
   * 返回成功消息
   *
   * @return 成功消息
   */
  public static AjaxResult success()
  {
    return AjaxResult.success("操作成功");
  }

  /**
   * 返回成功数据
   *
   * @return 成功消息
   */
  public static AjaxResult success(Object data)
  {
    return AjaxResult.success("操作成功", data);
  }

  /**
   * 返回成功消息
   *
   * @param msg 返回内容
   * @return 成功消息
   */
  public static AjaxResult success(String msg)
  {
    return AjaxResult.success(msg, null);
  }

  /**
   * 返回成功消息
   *
   * @return 成功消息
   */
  public static AjaxResult success(long total)
  {
    return AjaxResult.success("总数",total);
  }

  /**
   * 返回成功消息
   *
   * @param msg 返回内容
   * @param total 数据总数
   * @return 成功消息
   */
  public static AjaxResult success(String msg, long total) {
    return new AjaxResult(HttpStatus.SUCCESS, msg, total);
  }

  /**
   * 返回成功消息
   *
   * @param msg 返回内容
   * @param data 数据对象
   * @return 成功消息
   */
  public static AjaxResult success(String msg, Object data)
  {
    return new AjaxResult(HttpStatus.SUCCESS, msg, data);
  }

  /**
   * 返回成功消息
   *
   * @param msg 返回内容
   * @param data 数据对象
   * @return 成功消息
   */
  public static AjaxResult success(String msg, Object data, Object total) {
    return new AjaxResult(HttpStatus.SUCCESS, msg, data, total);
  }

  /**
   * 返回影响行数的结果
   * @param rows 影响行数
   * @return
   */
  public static AjaxResult toAjax(int rows){
    return rows > 0 ? AjaxResult.success() : AjaxResult.error();
  }

  /**
   * 返回影响行数的结果
   * @param result 结果
   * @return
   */
  public static AjaxResult toAjax(boolean result){
    return result ? AjaxResult.success() : AjaxResult.error();
  }

  /**
   * 返回错误消息
   *
   * @return
   */
  public static AjaxResult error()
  {
    return AjaxResult.error("操作失败");
  }

  /**
   * 返回错误消息
   *
   * @param msg 返回内容
   * @return 警告消息
   */
  public static AjaxResult error(String msg)
  {
    return AjaxResult.error(msg, null);
  }

  /**
   * 返回错误消息
   *
   * @param msg 返回内容
   * @param data 数据对象
   * @return 警告消息
   */
  public static AjaxResult error(String msg, Object data)
  {
    return new AjaxResult(HttpStatus.ERROR, msg, data);
  }

  /**
   * 返回错误消息
   *
   * @param code 状态码
   * @param msg 返回内容
   * @return 警告消息
   */
  public static AjaxResult error(int code, String msg)
  {
    return new AjaxResult(code, msg, null);
  }
}
